how write program in python